import os

def clear_screenshots():
    # 获取当前工作目录
    current_dir = os.getcwd()
    
    # 计数器，用于记录删除的文件数量
    deleted_count = 0
    
    # 遍历当前目录下的所有文件
    for filename in os.listdir(current_dir):
        # 检查文件名是否以 "screenshot" 开头且以 ".png" 结尾
        if filename.startswith("screenshot") and filename.endswith(".png"):
            try:
                # 构造完整文件路径
                file_path = os.path.join(current_dir, filename)
                # 删除文件
                os.remove(file_path)
                deleted_count += 1
                print(f"已删除: {filename}")
            except Exception as e:
                print(f"删除 {filename} 失败: {str(e)}")
    
    # 输出总结
    if deleted_count > 0:
        print(f"清理完成，共删除 {deleted_count} 个截图文件。")
    else:
        print("当前目录下没有找到符合条件的截图文件。")

if __name__ == "__main__":
    clear_screenshots()